import os
import cv2


def clear_dir(path):
    '''
    清空文件夹
    :param path:
    :return:
    '''
    img_list = get_all_file_in_dir(path)
    for img in img_list:
        # 此处可以增加文件名称过滤条件， 比如 -----
        # 跳过所有 __init__.py 文件
        if img == '__init__.py':
            continue
        # --------------------------------------
        os.remove(img)


def get_all_file_in_dir(dir_path, order='str'):
    '''
    获取目录下的所有文件
    :param dir_path:
    :param order: str(文件名以字符串排序), int（文件名以整数排序）
    :return:
    '''
    file_name_list = []
    for root, dirs, files in os.walk(dir_path):
        if files:
            for name in files:
                # 此处可以增加文件名称过滤条件， 比如 -----
                # 跳过所有 __init__.py 文件
                if name == '__init__.py':
                    continue
                # --------------------------------------

                file_name = '{0}{1}'.format(root, name).replace('\\', '/')
                file_name_list.append(file_name)

    if order == 'int':
        file_name_list.sort(key=lambda x: (int(x.replace(dir_path, '').split('.')[0])))

    return file_name_list


if __name__ == '__main__':
    # video_name = '../production_process/1_input_video/v111.mp4'
    # get_fps_from_video(video_path=video_name)

    img_list = get_all_file_in_dir('../production_process/2_split_imgs/')
    for im in img_list:
        print(im)